import CodeMirror from 'codemirror'

(function() {

	CodeMirror.extendMode('css', {
		commentStart: '/*',
		commentEnd: '*/',
		newlineAfterToken: function(type, content) {
			return /^[;{}]$/.test(content)
		}
	})

	CodeMirror.extendMode('javascript', {
		commentStart: '/*',
		commentEnd: '*/',
		// FIXME semicolons inside of for
		newlineAfterToken: function(type, content, textAfter, state) {
			if (this.jsonMode) {
				return /^[\[,{]$/.test(content) || /^}/.test(textAfter)
			} else {
				if (content == ';' && state.lexical && state.lexical.type == ')') return false
				return /^[;{}]$/.test(content) && !/^;/.test(textAfter)
			}
		}
	})

	CodeMirror.extendMode('xml', {
		commentStart: '<!--',
		commentEnd: '-->',
		newlineAfterToken: function(type, content, textAfter) {
			return type == 'tag' && />$/.test(content) || /^</.test(textAfter)
		}
	})

	// Comment/uncomment the specified range
	CodeMirror.defineExtension('commentRange', function(isComment, from, to) {
		var cm = this, curMode = CodeMirror.innerMode(cm.getMode(), cm.getTokenAt(from).state).mode
		cm.operation(function() {
			if (isComment) { // Comment range
				cm.replaceRange(curMode.commentEnd, to)
				cm.replaceRange(curMode.commentStart, from)
				if (from.line == to.line && from.ch == to.ch) // An empty comment inserted - put cursor inside
				{
					cm.setCursor(from.line, from.ch + curMode.commentStart.length)
				}
			} else { // Uncomment range
				var selText = cm.getRange(from, to)
				var startIndex = selText.indexOf(curMode.commentStart)
				var endIndex = selText.lastIndexOf(curMode.commentEnd)
				if (startIndex > -1 && endIndex > -1 && endIndex > startIndex) {
					// Take string till comment start
					selText = selText.substr(0, startIndex)
						// From comment start till comment end
						+ selText.substring(startIndex + curMode.commentStart.length, endIndex)
						// From comment end till string end
						+ selText.substr(endIndex + curMode.commentEnd.length)
				}
				cm.replaceRange(selText, from, to)
			}
		})
	})

	// Applies automatic mode-aware indentation to the specified range
	CodeMirror.defineExtension('autoIndentRange', function(from, to) {
		var cmInstance = this
		this.operation(function() {
			for (var i = from.line; i <= to.line; i++) {
				cmInstance.indentLine(i, 'smart')
			}
		})
	})

	// Applies automatic formatting to the specified range
	CodeMirror.defineExtension('autoFormatRange', function(from, to) {
		var cm = this
		var outer = cm.getMode(), text = cm.getRange(from, to).split('\n')
		var state = CodeMirror.copyState(outer, cm.getTokenAt(from).state)
		var tabSize = cm.getOption('tabSize')

		var out = '', lines = 0, atSol = from.ch == 0

		function newline() {
			out += '\n'
			atSol = true
			++lines
		}

		for (var i = 0; i < text.length; ++i) {
			var stream = new CodeMirror.StringStream(text[i], tabSize)
			while (!stream.eol()) {
				var inner = CodeMirror.innerMode(outer, state)
				var style = outer.token(stream, state), cur = stream.current()
				stream.start = stream.pos
				if (!atSol || /\S/.test(cur)) {
					out += cur
					atSol = false
				}
				if (!atSol && inner.mode.newlineAfterToken &&
					inner.mode.newlineAfterToken(style, cur, stream.string.slice(stream.pos) || text[i + 1] || '', inner.state)) {
					newline()
				}
			}
			if (!stream.pos && outer.blankLine) outer.blankLine(state)
			if (!atSol) newline()
		}

		cm.operation(function() {
			cm.replaceRange(out, from, to)
			for (var cur = from.line + 1, end = from.line + lines; cur <= end; ++cur) {
				cm.indentLine(cur, 'smart')
			}
			cm.setSelection(from, cm.getCursor(false))
		})
	})
	// 2021年12月8日 10点08分 新增，支持格式化所有 Applies automatic formatting to the specified range
	CodeMirror.defineExtension('autoFormatRangeAll', function() {
		var cm = this
		from = { line: editorPluginHtml.firstLine(), ch: 0, sticky: null }
		to = {
			line: editorPluginHtml.lastLine(),
			ch: editorPluginHtml.getLine(editorPluginHtml.lastLine()).length,
			sticky: null
		}
		var outer = cm.getMode(), text = cm.getRange(from, to).split('\n')
		var state = CodeMirror.copyState(outer, cm.getTokenAt(from).state)
		var tabSize = cm.getOption('tabSize')

		var out = '', lines = 0, atSol = from.ch == 0

		function newline() {
			out += '\n'
			atSol = true
			++lines
		}

		for (var i = 0; i < text.length; ++i) {
			var stream = new CodeMirror.StringStream(text[i], tabSize)
			while (!stream.eol()) {
				var inner = CodeMirror.innerMode(outer, state)
				var style = outer.token(stream, state), cur = stream.current()
				stream.start = stream.pos
				if (!atSol || /\S/.test(cur)) {
					out += cur
					atSol = false
				}
				if (!atSol && inner.mode.newlineAfterToken &&
					inner.mode.newlineAfterToken(style, cur, stream.string.slice(stream.pos) || text[i + 1] || '', inner.state)) {
					newline()
				}
			}
			if (!stream.pos && outer.blankLine) outer.blankLine(state)
			if (!atSol) newline()
		}

		cm.operation(function() {
			cm.replaceRange(out, from, to)
			for (var cur = from.line + 1, end = from.line + lines; cur <= end; ++cur) {
				cm.indentLine(cur, "smart");
			}
			cm.setSelection(from, cm.getCursor(false));
		});
	});
})();
